# coding=utf-8
# Copyright 2024 The Google Research Authors.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""Mean Map constants."""

PAIRS_LIST = [
    (1, 7),
    (1, 10),
    (2, 7),
    (2, 10),
    (2, 11),
    (2, 13),
    (3, 7),
    (3, 10),
    (3, 11),
    (3, 13),
    (4, 7),
    (4, 10),
    (4, 11),
    (4, 13),
    (4, 15),
    (6, 7),
    (6, 10),
    (7, 8),
    (7, 10),
    (7, 12),
    (7, 14),
    (7, 15),
    (7, 16),
    (7, 18),
    (7, 20),
    (7, 21),
    (7, 24),
    (7, 26),
    (10, 12),
    (10, 14),
    (10, 15),
    (10, 16),
    (10, 17),
    (10, 18),
    (10, 20),
    (10, 21),
    (10, 24),
    (10, 26),
    (11, 12),
    (11, 15),
    (11, 16),
    (11, 18),
    (11, 21),
    (11, 24),
    (11, 26),
    (12, 13),
    (13, 15),
    (13, 16),
    (13, 18),
    (13, 21),
    (13, 24),
    (13, 26),
]

OFFSETS = [
    0,
    75,
    233,
    355,
    402,
    689,
    857,
    976,
    1051,
    1156,
    1161,
    1190,
    1258,
    1337,
    2779,
    3332,
    179112,
    307620,
    307924,
    307942,
    319871,
    320499,
    320501,
    361724,
    366883,
    541717,
    544891,
    544917,
    556170,
    721375,
    721384,
    725988,
    728004,
    728007,
    900328,
    900345,
    900359,
    956814,
    956899,
]

FEATURE_COLS = [
    'I1',
    'I2',
    'I3',
    'I4',
    'I5',
    'I6',
    'I7',
    'I8',
    'I9',
    'I10',
    'I11',
    'I12',
    'I13',
    'C1',
    'C2',
    'C3',
    'C4',
    'C5',
    'C6',
    'C7',
    'C8',
    'C9',
    'C10',
    'C11',
    'C12',
    'C13',
    'C14',
    'C15',
    'C16',
    'C17',
    'C18',
    'C19',
    'C20',
    'C21',
    'C22',
    'C23',
    'C24',
    'C25',
    'C26',
]

LIST_SIZES = [64, 128, 256, 512]

LIST_NUM_SEGS = [40, 20, 10, 5]

TOTAL_NUM_SEGS = 40
